package com.boil.exception;

import com.boil.pojo.ResultVo;
import com.boil.util.Constants;
import com.boil.util.FastJsonUtils;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.apache.shiro.SecurityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.servlet.HandlerExceptionResolver;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;

/**
 * 
 * 全局异常处理机制。
 * 
 * @author ChenYong
 * @date 2016-09-21
 *
 */
public class GlobalHandlerExceptionResolver implements HandlerExceptionResolver {
	/** 日志 */
	private static Logger log = LoggerFactory.getLogger(GlobalHandlerExceptionResolver.class);

	/**
	 * 
	 * 处理异常。
	 * 
	 */
	@Override
	public ModelAndView resolveException(HttpServletRequest request, // 请求
			HttpServletResponse response, // 响应
			Object handler, // 处理机制
			Exception ex) { // 所有异常
		// 视图模型
		ModelAndView maView = new ModelAndView();

		// 请求参数异常
		if (ex instanceof RequestParamException) {
			RequestParamException ex1 = (RequestParamException) ex;

			// Ajax 请求
			if (Constants.AJAX_REQUEST.equalsIgnoreCase(request.getHeader(Constants.REQUEST_TYPE))) {
				log.error("Ajax 请求--->请求参数不合法");
				
				// 请求参数不合法：412
				response.setStatus(HttpServletResponse.SC_PRECONDITION_FAILED);
				
				maView.setViewName(null);
				
				PrintWriter pw = null;
				
				try {
					pw = response.getWriter();
					
					pw.write(FastJsonUtils.bean2Json(ex1.getResultVo()));
				} catch (IOException e) {
					log.error("Ajax 请求--->响应请求参数不合法时出现异常，异常信息如下：{}", e);
				} finally {
					try {
						if (pw != null) {
							pw.close();
						}
					} catch (Exception e) {
						log.error("Ajax 请求--->响应请求参数不合法，关闭响应流时出现异常，异常信息如下：{}", e);
					}
				}
				
				// 非 Ajax 请求
			} else {
				// 请求来自 Uploadifive 文件上传
				if (Constants.UPLOADIFIVE.equals(request.getParameter(Constants.UPLOADIFIVE))) {
					log.error("Uploadifive 文件上传请求--->请求参数不合法");
					
					// 请求参数不合法：412
					// response.setStatus(HttpServletResponse.SC_PRECONDITION_FAILED);
					
					maView.setViewName(null);
					
					PrintWriter pw = null;
					
					try {
						pw = response.getWriter();
						
						pw.write(FastJsonUtils.bean2Json(ex1.getResultVo()));
					} catch (IOException e) {
						log.error("Uploadifive 文件上传请求--->响应请求参数不合法时出现异常，异常信息如下：{}", e);
					} finally {
						try {
							if (pw != null) {
								pw.close();
							}
						} catch (Exception e) {
							log.error("Uploadifive 文件上传--->响应请求参数不合法，关闭响应流时出现异常，异常信息如下：{}", e);
						}
					}
					
					// 普通请求
				} else {
					log.error("非 Ajax 请求--->请求参数不合法");

					SecurityUtils.getSubject().getSession().setAttribute(Constants.ERROR_412_STR, ex1.getResultVo());
					
					maView.setViewName("redirect:/web/error/412.html");
				}
			}

			// 未授权访问
		} else if (ex instanceof UnauthorizedException) {
			// 结果
			ResultVo resultVo = new ResultVo();
			
			resultVo.setResult(ExceptionUtils.getStackTrace(ex));
			resultVo.setResultCode(Constants.UNAUTHORIZED_ERROR);
			resultVo.setResultMsg(Constants.UNAUTHORIZED_ERROR_MSG);
			
			// Ajax 请求
			if (Constants.AJAX_REQUEST.equalsIgnoreCase(request.getHeader(Constants.REQUEST_TYPE))) {
				log.error("Ajax 请求--->未授权访问");
				
				// 未授权访问：403
				response.setStatus(HttpServletResponse.SC_FORBIDDEN);
				
				maView.setViewName(null);
				
				PrintWriter pw = null;
				
				try {
					pw = response.getWriter();
					
					pw.write(FastJsonUtils.bean2Json(resultVo));
				} catch (IOException e) {
					log.error("Ajax 请求--->响应未授权访问时出现异常，异常信息如下：{}", e);
				} finally {
					try {
						if (pw != null) {
							pw.close();
						}
					} catch (Exception e) {
						log.error("Ajax 请求--->响应未授权访问，关闭响应流时出现异常，异常信息如下：{}", e);
					}
				}
				
				// 非 Ajax 请求
			} else {
				// 请求来自 Uploadifive 文件上传
				if (Constants.UPLOADIFIVE.equals(request.getParameter(Constants.UPLOADIFIVE))) {
					log.error("Uploadifive 文件上传请求--->未授权访问");
					
					// 未授权访问：403
					// response.setStatus(HttpServletResponse.SC_FORBIDDEN);
					
					maView.setViewName(null);
					
					PrintWriter pw = null;
					
					try {
						pw = response.getWriter();
						
						pw.write(FastJsonUtils.bean2Json(resultVo));
					} catch (IOException e) {
						log.error("Uploadifive 文件上传请求--->响应未授权访问时出现异常，异常信息如下：{}", e);
					} finally {
						try {
							if (pw != null) {
								pw.close();
							}
						} catch (Exception e) {
							log.error("Uploadifive 文件上传请求--->响应未授权访问，关闭响应流时出现异常，异常信息如下：{}", e);
						}
					}
					
					// 普通请求
				} else {
					log.error("非 Ajax 请求--->未授权访问");
					
					SecurityUtils.getSubject().getSession().setAttribute(Constants.ERROR_403_STR, resultVo);
					
					maView.setViewName("redirect:/web/error/403.html");
				}
			}

			log.error("未授权访问异常：{}", ex);

			// 服务器内部异常
		} else {
			// 结果
			ResultVo resultVo = new ResultVo();
			
			resultVo.setResult(ExceptionUtils.getStackTrace(ex));
			resultVo.setResultCode(Constants.SYS_ERROR);
			resultVo.setResultMsg(Constants.SYS_ERROR_MSG);
			
			// Ajax 请求
			if (Constants.AJAX_REQUEST.equalsIgnoreCase(request.getHeader(Constants.REQUEST_TYPE))) {
				log.error("Ajax 请求--->服务器内部异常");
				
				// 服务器内部异常：500
				response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
				
				maView.setViewName(null);
				
				PrintWriter pw = null;
				
				try {
					pw = response.getWriter();
					
					pw.write(FastJsonUtils.bean2Json(resultVo));
				} catch (IOException e) {
					log.error("Ajax 请求--->响应服务器内部异常时出现异常，异常信息如下：{}", e);
				} finally {
					try {
						if (pw != null) {
							pw.close();
						}
					} catch (Exception e) {
						log.error("Ajax 请求--->响应服务器内部异常，关闭响应流时出现异常，异常信息如下：{}", e);
					}
				}
				
				// 非 Ajax 请求
			} else {
				// 请求来自 Uploadifive 文件上传
				if (Constants.UPLOADIFIVE.equals(request.getParameter(Constants.UPLOADIFIVE))) {
					log.error("Uploadifive 文件上传请求--->服务器内部异常");
					
					// 服务器内部异常：500
					// response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
					
					maView.setViewName(null);
					
					PrintWriter pw = null;
					
					try {
						pw = response.getWriter();
						
						pw.write(FastJsonUtils.bean2Json(resultVo));
					} catch (IOException e) {
						log.error("Uploadifive 文件上传请求--->响应服务器内部异常时出现异常，异常信息如下：{}", e);
					} finally {
						try {
							if (pw != null) {
								pw.close();
							}
						} catch (Exception e) {
							log.error("Uploadifive 文件上传请求--->响应服务器内部异常，关闭响应流时出现异常，异常信息如下：{}", e);
						}
					}
					
					// 普通请求
				} else {
					log.error("非 Ajax 请求--->服务器内部异常");
					
					SecurityUtils.getSubject().getSession().setAttribute(Constants.ERROR_500_STR, resultVo);
					
					maView.setViewName("redirect:/web/error/500.html");
				}
			}

			log.error("服务器内部异常：{}", ex);
		}

		return maView;
	}
}